import torch

# 检查 PyTorch 版本
print(f"PyTorch version: {torch.__version__}")

# 检查 CUDA 版本 (如果安装了GPU版PyTorch)
if torch.cuda.is_available():
    print(f"CUDA version: {torch.version.cuda}")
else:
    print("CUDA not available, using CPU version.")

# 作者2也提交了一些功能上去了

# 检查是否可以使用 GPU
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(f"Using device: {device}")
# 创建一个简单的张量并将其移动到 GPU (如果可用)
x = torch.tensor([1.0, 2.0, 3.0])
x = x.to(device)
print(f"Tensor on device: {x}")



# 我先提交一些功能上去了

print(2)


#作者3同步开发
print("author3,为什么没成功")

# 可以本地合并然后推送
